<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">

<html lang="en">

<head>
  <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
  <title>LCOV - code analysis - src/caffe/layers/base_data_layer.cpp</title>
  <link rel="stylesheet" type="text/css" href="../../../gcov.css">
</head>

<body>

  <table width="100%" border=0 cellspacing=0 cellpadding=0>
    <tr><td class="title">LCOV - code coverage report</td></tr>
    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>

    <tr>
      <td width="100%">
        <table cellpadding=1 border=0 width="100%">
          <tr>
            <td width="10%" class="headerItem">Current view:</td>
            <td width="35%" class="headerValue"><a href="../../../index.html">top level</a> - <a href="index.html">src/caffe/layers</a> - base_data_layer.cpp<span style="font-size: 80%;"> (source / <a href="base_data_layer.cpp.func-sort-c.html">functions</a>)</span></td>
            <td width="5%"></td>
            <td width="15%"></td>
            <td width="10%" class="headerCovTableHead">Hit</td>
            <td width="10%" class="headerCovTableHead">Total</td>
            <td width="15%" class="headerCovTableHead">Coverage</td>
          </tr>
          <tr>
            <td class="headerItem">Test:</td>
            <td class="headerValue">code analysis</td>
            <td></td>
            <td class="headerItem">Lines:</td>
            <td class="headerCovTableEntry">43</td>
            <td class="headerCovTableEntry">45</td>
            <td class="headerCovTableEntryHi">95.6 %</td>
          </tr>
          <tr>
            <td class="headerItem">Date:</td>
            <td class="headerValue">2020-09-11 22:25:26</td>
            <td></td>
            <td class="headerItem">Functions:</td>
            <td class="headerCovTableEntry">8</td>
            <td class="headerCovTableEntry">16</td>
            <td class="headerCovTableEntryLo">50.0 %</td>
          </tr>
          <tr>
            <td class="headerItem">Legend:</td>
            <td class="headerValueLeg">            Lines:
            <span class="coverLegendCov">hit</span>
            <span class="coverLegendNoCov">not hit</span>
</td>
            <td></td>
          </tr>
          <tr><td><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
        </table>
      </td>
    </tr>

    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
  </table>

  <table cellpadding=0 cellspacing=0 border=0>
    <tr>
      <td><br></td>
    </tr>
    <tr>
      <td>
<pre class="sourceHeading">          Line data    Source code</pre>
<pre class="source">
<a name="1"><span class="lineNum">       1 </span>            : #include &lt;boost/thread.hpp&gt;</a>
<span class="lineNum">       2 </span>            : #include &lt;vector&gt;
<span class="lineNum">       3 </span>            : 
<span class="lineNum">       4 </span>            : #include &quot;caffe/blob.hpp&quot;
<span class="lineNum">       5 </span>            : #include &quot;caffe/data_transformer.hpp&quot;
<span class="lineNum">       6 </span>            : #include &quot;caffe/internal_thread.hpp&quot;
<span class="lineNum">       7 </span>            : #include &quot;caffe/layer.hpp&quot;
<span class="lineNum">       8 </span>            : #include &quot;caffe/layers/base_data_layer.hpp&quot;
<span class="lineNum">       9 </span>            : #include &quot;caffe/proto/caffe.pb.h&quot;
<span class="lineNum">      10 </span>            : #include &quot;caffe/util/blocking_queue.hpp&quot;
<span class="lineNum">      11 </span>            : 
<span class="lineNum">      12 </span>            : namespace caffe {
<a name="13"><span class="lineNum">      13 </span>            : </a>
<span class="lineNum">      14 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">      15 </span><span class="lineCov">          1 : BaseDataLayer&lt;Dtype&gt;::BaseDataLayer(const LayerParameter&amp; param)</span>
<span class="lineNum">      16 </span>            :     : Layer&lt;Dtype&gt;(param),
<span class="lineNum">      17 </span><span class="lineCov">          2 :       transform_param_(param.transform_param()) {</span>
<span class="lineNum">      18 </span><span class="lineCov">          1 : }</span>
<a name="19"><span class="lineNum">      19 </span>            : </a>
<span class="lineNum">      20 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">      21 </span><span class="lineCov">          1 : void BaseDataLayer&lt;Dtype&gt;::LayerSetUp(const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; bottom,</span>
<span class="lineNum">      22 </span>            :       const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; top) {
<span class="lineNum">      23 </span><span class="lineCov">          1 :   if (top.size() == 1) {</span>
<span class="lineNum">      24 </span><span class="lineNoCov">          0 :     output_labels_ = false;</span>
<span class="lineNum">      25 </span>            :   } else {
<span class="lineNum">      26 </span><span class="lineCov">          1 :     output_labels_ = true;</span>
<span class="lineNum">      27 </span>            :   }
<span class="lineNum">      28 </span><span class="lineCov">          1 :   data_transformer_.reset(</span>
<span class="lineNum">      29 </span>            :       new DataTransformer&lt;Dtype&gt;(transform_param_, this-&gt;phase_));
<span class="lineNum">      30 </span><span class="lineCov">          1 :   data_transformer_-&gt;InitRand();</span>
<span class="lineNum">      31 </span>            :   // The subclasses should setup the size of bottom and top
<span class="lineNum">      32 </span><span class="lineCov">          1 :   DataLayerSetUp(bottom, top);</span>
<span class="lineNum">      33 </span><span class="lineCov">          1 : }</span>
<a name="34"><span class="lineNum">      34 </span>            : </a>
<span class="lineNum">      35 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">      36 </span><span class="lineCov">          1 : BasePrefetchingDataLayer&lt;Dtype&gt;::BasePrefetchingDataLayer(</span>
<span class="lineNum">      37 </span>            :     const LayerParameter&amp; param)
<span class="lineNum">      38 </span>            :     : BaseDataLayer&lt;Dtype&gt;(param),
<span class="lineNum">      39 </span>            :       prefetch_(param.data_param().prefetch()),
<span class="lineNum">      40 </span><span class="lineCov">          4 :       prefetch_free_(), prefetch_full_(), prefetch_current_() {</span>
<span class="lineNum">      41 </span><span class="lineCov">         14 :   for (int i = 0; i &lt; prefetch_.size(); ++i) {</span>
<span class="lineNum">      42 </span><span class="lineCov">          8 :     prefetch_[i].reset(new Batch&lt;Dtype&gt;());</span>
<span class="lineNum">      43 </span><span class="lineCov">          4 :     prefetch_free_.push(prefetch_[i].get());</span>
<span class="lineNum">      44 </span>            :   }
<span class="lineNum">      45 </span><span class="lineCov">          1 : }</span>
<a name="46"><span class="lineNum">      46 </span>            : </a>
<span class="lineNum">      47 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">      48 </span><span class="lineCov">          1 : void BasePrefetchingDataLayer&lt;Dtype&gt;::LayerSetUp(</span>
<span class="lineNum">      49 </span>            :     const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; bottom, const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; top) {
<span class="lineNum">      50 </span><span class="lineCov">          1 :   BaseDataLayer&lt;Dtype&gt;::LayerSetUp(bottom, top);</span>
<span class="lineNum">      51 </span>            : 
<span class="lineNum">      52 </span>            :   // Before starting the prefetch thread, we make cpu_data and gpu_data
<span class="lineNum">      53 </span>            :   // calls so that the prefetch thread does not accidentally make simultaneous
<span class="lineNum">      54 </span>            :   // cudaMalloc calls when the main thread is running. In some GPUs this
<span class="lineNum">      55 </span>            :   // seems to cause failures if we do not so.
<span class="lineNum">      56 </span><span class="lineCov">         14 :   for (int i = 0; i &lt; prefetch_.size(); ++i) {</span>
<span class="lineNum">      57 </span><span class="lineCov">          4 :     prefetch_[i]-&gt;data_.mutable_cpu_data();</span>
<span class="lineNum">      58 </span><span class="lineCov">          4 :     if (this-&gt;output_labels_) {</span>
<span class="lineNum">      59 </span><span class="lineCov">          4 :       prefetch_[i]-&gt;label_.mutable_cpu_data();</span>
<span class="lineNum">      60 </span>            :     }
<span class="lineNum">      61 </span>            :   }
<span class="lineNum">      62 </span>            : #ifndef CPU_ONLY
<span class="lineNum">      63 </span>            :   if (Caffe::mode() == Caffe::GPU) {
<span class="lineNum">      64 </span>            :     for (int i = 0; i &lt; prefetch_.size(); ++i) {
<span class="lineNum">      65 </span>            :       prefetch_[i]-&gt;data_.mutable_gpu_data();
<span class="lineNum">      66 </span>            :       if (this-&gt;output_labels_) {
<span class="lineNum">      67 </span>            :         prefetch_[i]-&gt;label_.mutable_gpu_data();
<span class="lineNum">      68 </span>            :       }
<span class="lineNum">      69 </span>            :     }
<span class="lineNum">      70 </span>            :   }
<span class="lineNum">      71 </span>            : #endif
<span class="lineNum">      72 </span>            :   DLOG(INFO) &lt;&lt; &quot;Initializing prefetch&quot;;
<span class="lineNum">      73 </span><span class="lineCov">          1 :   this-&gt;data_transformer_-&gt;InitRand();</span>
<span class="lineNum">      74 </span><span class="lineCov">          1 :   StartInternalThread();</span>
<span class="lineNum">      75 </span>            :   DLOG(INFO) &lt;&lt; &quot;Prefetch initialized.&quot;;
<span class="lineNum">      76 </span><span class="lineCov">          1 : }</span>
<a name="77"><span class="lineNum">      77 </span>            : </a>
<span class="lineNum">      78 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">      79 </span><span class="lineCov">          1 : void BasePrefetchingDataLayer&lt;Dtype&gt;::InternalThreadEntry() {</span>
<span class="lineNum">      80 </span>            : #ifndef CPU_ONLY
<span class="lineNum">      81 </span>            :   cudaStream_t stream;
<span class="lineNum">      82 </span>            :   if (Caffe::mode() == Caffe::GPU) {
<span class="lineNum">      83 </span>            :     CUDA_CHECK(cudaStreamCreateWithFlags(&amp;stream, cudaStreamNonBlocking));
<span class="lineNum">      84 </span>            :   }
<span class="lineNum">      85 </span>            : #endif
<span class="lineNum">      86 </span>            : 
<span class="lineNum">      87 </span>            :   try {
<span class="lineNum">      88 </span><span class="lineCov">        207 :     while (!must_stop()) {</span>
<span class="lineNum">      89 </span><span class="lineCov">        208 :       Batch&lt;Dtype&gt;* batch = prefetch_free_.pop();</span>
<span class="lineNum">      90 </span><span class="lineCov">        103 :       load_batch(batch);</span>
<span class="lineNum">      91 </span>            : #ifndef CPU_ONLY
<span class="lineNum">      92 </span>            :       if (Caffe::mode() == Caffe::GPU) {
<span class="lineNum">      93 </span>            :         batch-&gt;data_.data().get()-&gt;async_gpu_push(stream);
<span class="lineNum">      94 </span>            :         if (this-&gt;output_labels_) {
<span class="lineNum">      95 </span>            :           batch-&gt;label_.data().get()-&gt;async_gpu_push(stream);
<span class="lineNum">      96 </span>            :         }
<span class="lineNum">      97 </span>            :         CUDA_CHECK(cudaStreamSynchronize(stream));
<span class="lineNum">      98 </span>            :       }
<span class="lineNum">      99 </span>            : #endif
<span class="lineNum">     100 </span><span class="lineCov">        103 :       prefetch_full_.push(batch);</span>
<span class="lineNum">     101 </span>            :     }
<span class="lineNum">     102 </span><span class="lineCov">          1 :   } catch (boost::thread_interrupted&amp;) {</span>
<span class="lineNum">     103 </span>            :     // Interrupted exception is expected on shutdown
<span class="lineNum">     104 </span>            :   }
<span class="lineNum">     105 </span>            : #ifndef CPU_ONLY
<span class="lineNum">     106 </span>            :   if (Caffe::mode() == Caffe::GPU) {
<span class="lineNum">     107 </span>            :     CUDA_CHECK(cudaStreamDestroy(stream));
<span class="lineNum">     108 </span>            :   }
<span class="lineNum">     109 </span>            : #endif
<span class="lineNum">     110 </span><span class="lineCov">          1 : }</span>
<a name="111"><span class="lineNum">     111 </span>            : </a>
<span class="lineNum">     112 </span>            : template &lt;typename Dtype&gt;
<span class="lineNum">     113 </span><span class="lineCov">        100 : void BasePrefetchingDataLayer&lt;Dtype&gt;::Forward_cpu(</span>
<span class="lineNum">     114 </span>            :     const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; bottom, const vector&lt;Blob&lt;Dtype&gt;*&gt;&amp; top) {
<span class="lineNum">     115 </span><span class="lineCov">        100 :   if (prefetch_current_) {</span>
<span class="lineNum">     116 </span><span class="lineCov">         99 :     prefetch_free_.push(prefetch_current_);</span>
<span class="lineNum">     117 </span>            :   }
<span class="lineNum">     118 </span><span class="lineCov">        200 :   prefetch_current_ = prefetch_full_.pop(&quot;Waiting for data&quot;);</span>
<span class="lineNum">     119 </span>            :   // Reshape to loaded data.
<span class="lineNum">     120 </span><span class="lineCov">        100 :   top[0]-&gt;ReshapeLike(prefetch_current_-&gt;data_);</span>
<span class="lineNum">     121 </span><span class="lineCov">        100 :   top[0]-&gt;set_cpu_data(prefetch_current_-&gt;data_.mutable_cpu_data());</span>
<span class="lineNum">     122 </span><span class="lineCov">        100 :   if (this-&gt;output_labels_) {</span>
<span class="lineNum">     123 </span>            :     // Reshape to loaded labels.
<span class="lineNum">     124 </span><span class="lineCov">        100 :     top[1]-&gt;ReshapeLike(prefetch_current_-&gt;label_);</span>
<span class="lineNum">     125 </span><span class="lineCov">        100 :     top[1]-&gt;set_cpu_data(prefetch_current_-&gt;label_.mutable_cpu_data());</span>
<span class="lineNum">     126 </span>            :   }
<span class="lineNum">     127 </span><span class="lineCov">        100 : }</span>
<a name="128"><span class="lineNum">     128 </span>            : </a>
<span class="lineNum">     129 </span>            : #ifdef CPU_ONLY
<span class="lineNum">     130 </span><span class="lineNoCov">          0 : STUB_GPU_FORWARD(BasePrefetchingDataLayer, Forward);</span>
<span class="lineNum">     131 </span>            : #endif
<span class="lineNum">     132 </span>            : 
<span class="lineNum">     133 </span>            : INSTANTIATE_CLASS(BaseDataLayer);
<a name="134"><span class="lineNum">     134 </span>            : INSTANTIATE_CLASS(BasePrefetchingDataLayer);</a>
<span class="lineNum">     135 </span>            : 
<span class="lineNum">     136 </span><span class="lineCov">          3 : }  // namespace caffe</span>
</pre>
      </td>
    </tr>
  </table>
  <br>

  <table width="100%" border=0 cellspacing=0 cellpadding=0>
    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
    <tr><td class="versionInfo">Generated by: <a href="http://ltp.sourceforge.net/coverage/lcov.php" target="_parent">LCOV version 1.12</a></td></tr>
  </table>
  <br>

</body>
</html>
